package com.atguigu.lease.common.converter;

import com.atguigu.lease.model.enums.BaseEnum;
import org.springframework.core.convert.converter.Converter;
import org.springframework.core.convert.converter.ConverterFactory;
import org.springframework.stereotype.Component;

/**
 * projectName: com.atguigu.lease.common.converter
 *
 * @author: wangs
 * description:
 */
@Component
public class StringToBaseEnumConverter implements ConverterFactory<String, BaseEnum> {

    @Override
    public <T extends BaseEnum> Converter<String, T> getConverter(Class<T> targetType) {
        return new Converter<String, T>() {
            @Override
            public T convert(String source) {
                 // 获取枚举实例
                T[] enumConstants = targetType.getEnumConstants();
                // 遍历
                for (T enumConstant : enumConstants) {
                    if(enumConstant.getCode().equals(Integer.parseInt(source))) {
                        return enumConstant;
                    }
                }
                throw new IllegalArgumentException("非法参数");
            }
        };
    }
}
